package com.harmony.pilot.agent.model;

import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.util.ArrayList;
import java.util.List;

@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class ConversationContext {
    private String projectRoot;
    private List<String> openFiles;
    private String currentFile;
    private CursorPosition cursorPosition;
    private List<CodeChange> recentChanges;
    private List<FileInfo> projectFiles;
    
    public ConversationContext() {
        this.openFiles = new ArrayList<>();
        this.recentChanges = new ArrayList<>();
        this.projectFiles = new ArrayList<>();
    }
    
    @Data
    @Builder
    @NoArgsConstructor
    @AllArgsConstructor
    public static class CursorPosition {
        private int line;
        private int column;
    }
    
    @Data
    @Builder
    @NoArgsConstructor
    @AllArgsConstructor
    public static class CodeChange {
        private String filePath;
        private String type; // INSERT, DELETE, MODIFY
        private String content;
        private int line;
    }
    
    @Data
    @Builder
    @NoArgsConstructor
    @AllArgsConstructor
    public static class FileInfo {
        private String path;
        private String type;
        private long size;
    }
}

